import argparse


opts = argparse.ArgumentParser()
opts.add_argument('--epochs', type=int, default=10)
opts.add_argument('--batch_size', type=int, default=64)
opts.add_argument('--device', type=str, default='cuda')
opts.add_argument('--img_size', type=int, default=32)
opts = opts.parse_args()